package com.hc.programming.array;

import java.util.*;

/**
 * 49.给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
 * <p>
 * 字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
 * <p>
 * 示例 1:
 * 输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
 * 输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
 * 示例 2:
 * 输入: strs = [""]
 * 输出: [[""]]
 * 示例 3:
 * 输入: strs = ["a"]
 * 输出: [["a"]]
 * <p>
 * 提示：
 * 1 <= strs.length <= 104
 * 0 <= strs[i].length <= 100
 * strs[i] 仅包含小写字母
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/7/7 17:15
 */
public class 字母异位词分组 {
    public static void main(String[] args) {
        System.out.println(groupAnagrams3(new String[]{"eat", "tea", "tan", "ate", "nat", "bat"}));
    }

    /**
     * 方案1：排序
     *
     * @param strs
     * @return
     */
    public static List<List<String>> groupAnagrams(String[] strs) {
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] arr = str.toCharArray();
            Arrays.sort(arr);
            map.computeIfAbsent(new String(arr), k -> new ArrayList<>()).add(str);
        }
        return new ArrayList<>(map.values());
    }

    /**
     * 方案2：字符映射
     *
     * @param strs
     * @return
     */
    public static List<List<String>> groupAnagrams2(String[] strs) {
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            int[] arr = new int[26];
            for (int i = 0; i < str.length(); i++) {
                arr[str.charAt(i) - 'a']++;
            }
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < arr.length; i++) {
                if (arr[i] > 0) {
                    sb.append('a' + i).append(arr[i]);
                }
            }
            map.computeIfAbsent(sb.toString(), k -> new ArrayList<>()).add(str);
        }
        return new ArrayList<>(map.values());
    }

    public static List<List<String>> groupAnagrams3(String[] strs) {
        return new AbstractList<List<String>>() {
            private final List<List<String>> list = new ArrayList<>();

            @Override
            public List<String> get(int index) {
                initList();
                return list.get(index);
            }

            private void initList() {
                // 初始化List，如果List不为空直接返回
                if (!list.isEmpty()) {
                    return;
                }
                Map<String, List<String>> map = new HashMap<>();
                for (String str : strs) {
                    char[] arr = str.toCharArray();
                    Arrays.sort(arr);
                    map.computeIfAbsent(new String(arr), k -> new ArrayList<>()).add(str);
                }
                list.addAll(map.values());
            }

            @Override
            public int size() {
                initList();
                return list.size();
            }
        };
    }
}
